package com.kanq.service;

import com.baomidou.mybatisplus.extension.service.IService;
import com.kanq.pojo.entity.Llm;
import com.kanq.pojo.request.ChatRequest;
import org.springframework.web.multipart.MultipartFile;
import org.springframework.web.servlet.mvc.method.annotation.SseEmitter;

import java.util.concurrent.ConcurrentHashMap;

public interface LLMService extends IService<Llm> {
    int readFile(MultipartFile file);

    String chat(String prompt);

    SseEmitter sseChat(ChatRequest request, ConcurrentHashMap<String, SseEmitter> taskEmitterMap);

    void stopChat(String taskId, ConcurrentHashMap<String, SseEmitter> taskEmitterMap);
}
